package com.example.springai.domain;


import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.ai.chat.messages.Message;

@NoArgsConstructor
@Data
public class MessageVO {
     private String role;
     private String content;

      public MessageVO(Message message) {
          switch ( message.getMessageType()){
              case USER:
                  this.role = "user";
                  break;
              case ASSISTANT:
                  this.role = "assistant";
                  break;
              default:
                  this.role = "";
                  break;
          }
         this.content = message.getText();
     }
}
